package pri.damai.xiaowu.common.log.trace.dubbo;

import org.apache.commons.lang3.StringUtils;
import org.apache.dubbo.common.constants.CommonConstants;
import org.apache.dubbo.common.extension.Activate;
import org.apache.dubbo.rpc.*;
import org.slf4j.MDC;
import pri.damai.xiaowu.common.log.trace.ThreadMdcUtil;

import static pri.damai.xiaowu.common.core.constant.TraceConstant.TRACE_ID;

/**
 *
 * @Desc
 * @Author DaMai
 * @Date 2021/5/26 9:57
 * 但行好事，莫问前程。
 */
@Activate(group = {CommonConstants.PROVIDER})
public class ProviderRpcTraceFilter implements Filter {

    @Override
    public Result invoke(Invoker<?> invoker, Invocation invocation) throws RpcException {
        // 接收消费端的traceId
        String traceId = RpcContext.getContext().getAttachment(TRACE_ID);
        if (StringUtils.isBlank(traceId)) {
            traceId = ThreadMdcUtil.createTraceId();
        }

        // 设置日志traceId
        ThreadMdcUtil.setTraceId(traceId);

        // 如果这个服务还会调用下一个服务，需要再次设置下游参数
        RpcContext.getContext().setAttachment(TRACE_ID, traceId);

        try {
            return invoker.invoke(invocation);
        } finally {
            // 调用完成后移除MDC属性
            MDC.remove(TRACE_ID);
        }
    }
}
